package com.xk.utils;

import cn.hutool.json.JSONObject;

import java.util.ArrayList;
import java.util.List;

public class AnswerSheetUtil {

    /**
     * 获取答题卡的id转为josn格式:{ "singleChoice":[1, 2, 3], "multipleChoice":[22, 33], "ifQuestions":[333, 444]}
     *
     * @param singleChoiceIds  单选题id集合
     * @param multipleChoiceIds 多选题id集合
     * @param ifQuestionIds 判断题id集合
     * @return JSON 格式字符串
     */
    private String getAnswerSheetIdsJson(List<Long> singleChoiceIds, List<Long> multipleChoiceIds, List<Long> ifQuestionIds) {
        // 创建 JSON 对象
        JSONObject jsonObject = new JSONObject();

        // 将各个集合放入对应的 JSON 字段中
        jsonObject.put("singleChoice", singleChoiceIds);
        jsonObject.put("multipleChoice", multipleChoiceIds);
        jsonObject.put("ifQuestions", ifQuestionIds);

        // 返回 JSON 字符串
        return jsonObject.toString();
    }

    public static void main(String[] args) {
        // 示例调用
        List<Long> singleChoiceIds = new ArrayList<>();
        singleChoiceIds.add(1L);
        singleChoiceIds.add(2L);
        singleChoiceIds.add(3L);

        List<Long> multipleChoiceIds = new ArrayList<>();
        multipleChoiceIds.add(22L);
        multipleChoiceIds.add(33L);
        List<Long> ifQuestionIds = new ArrayList<>();
        ifQuestionIds.add(333L);
        ifQuestionIds.add(444L);

        AnswerSheetUtil util = new AnswerSheetUtil();
        String json = util.getAnswerSheetIdsJson(singleChoiceIds, multipleChoiceIds, ifQuestionIds);

        System.out.println(json);
    }
}
